// Copyright 2024-2025 Golem Cloud
//
// Licensed under the Golem Source License v1.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://license.golem.cloud/LICENSE
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use crate::durable_host::{Durability, DurableWorkerCtx};
use crate::workerctx::WorkerCtx;
use golem_common::model::oplog::types::SerializableDateTime;
use golem_common::model::oplog::{
    host_functions, DurableFunctionType, HostRequestNoInput, HostResponseWallClock,
};
use wasmtime_wasi::p2::bindings::clocks::wall_clock::{Datetime, Host};

impl<Ctx: WorkerCtx> Host for DurableWorkerCtx<Ctx> {
    async fn now(&mut self) -> anyhow::Result<Datetime> {
        let durability =
            Durability::<host_functions::WallClockNow>::new(self, DurableFunctionType::ReadLocal)
                .await?;

        let result = if durability.is_live() {
            let result = Host::now(&mut self.as_wasi_view()).await?;

            durability
                .persist(
                    self,
                    HostRequestNoInput {},
                    HostResponseWallClock {
                        time: SerializableDateTime {
                            seconds: result.seconds,
                            nanoseconds: result.nanoseconds,
                        },
                    },
                )
                .await
        } else {
            durability.replay(self).await
        }?;

        Ok(result.time.into())
    }

    async fn resolution(&mut self) -> anyhow::Result<Datetime> {
        let durability = Durability::<host_functions::WallClockResolution>::new(
            self,
            DurableFunctionType::ReadLocal,
        )
        .await?;

        let result = if durability.is_live() {
            let result = Host::resolution(&mut self.as_wasi_view()).await?;
            durability
                .persist(
                    self,
                    HostRequestNoInput {},
                    HostResponseWallClock {
                        time: SerializableDateTime {
                            seconds: result.seconds,
                            nanoseconds: result.nanoseconds,
                        },
                    },
                )
                .await
        } else {
            durability.replay(self).await
        }?;

        Ok(result.time.into())
    }
}
